import numpy as np

us_data = "./youtube_video_data/US_video_data_numbers.csv"
uk_data = "./youtube_video_data/GB_video_data_numbers.csv"

'''
现在希望把之前案例中两个国家的数据方法一起来研究分析，同时保留国家的信息（每条数据的国家来源），应该怎么办
'''

# 加载国家数据
us_data = np.loadtxt(us_data, delimiter=",", dtype=int)
uk_data = np.loadtxt(uk_data, delimiter=",", dtype=int)

# 添加国家信息
# zeros 构造全为0的数据  0代表us，1代表uk
zeros_data = np.zeros((us_data.shape[0], 1)).astype(int)  # shape 查看数组的形状
print("zeros_data=", zeros_data)
ones_data = np.ones((uk_data.shape[0], 1)).astype(int)
print("ones_data=", ones_data)

# 分别添加一列全为0,1的数组
us_data = np.hstack((us_data, zeros_data))  # 水平拼接
uk_data = np.hstack((uk_data, ones_data))

# 拼接两组数据
final_data = np.vstack(((us_data, uk_data)))  # 竖直拼接
print(final_data)
